package com.Hsu.mq.mqserver.core;

/**
 * @author Hsu琛君珩
 * @ClassName:MSGQueue
 * @date 2024-02-09
 * @apiNote
 * @Version: v1.0
 */

import com.Hsu.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 存储消息的队列
 */
@Data
public class MSGQueue {
    //表示队列的身份标识
    private String name;
    //表示队列是否持久化
    private boolean durable=false;
    //表示队列是否独有的，true 表示只能被一个消费者使用，false 则大家都能用。不过我们还没实现
    private boolean exclusive=false;
    //表示队列没消费者使用时是否自动删除。不过我们还没实现
    private boolean autoDelete=false;
    //表示队列的扩展参数。不过我们还没实现
    private Map<String,Object> arguments=new HashMap<>();

    //扩展属性
    //当前队列都有哪些消费者订阅了
    private List<ConsumerEnv> consumerEnvList=new ArrayList<>();
    //记录当前取到了第几个消费者，方便实现轮询策略
    private AtomicInteger consumerSeq=new AtomicInteger();//方便多线程

    /**
     * 添加一个新的订阅者
     * @param consumerEnv
     */
    public void addConsumerEnv(ConsumerEnv consumerEnv){
        //synchronized (this){//加锁
            consumerEnvList.add(consumerEnv);
        //}
    }

    //订阅者的删除先不考虑

    /**
     * 挑选一个订阅者，用来处理当前的消息（按照轮询的方式）
     * @return
     */
    public ConsumerEnv chooseConsumer() {
        if (consumerEnvList.size() == 0) {
            // 该队列没有人订阅的
            return null;
        }
        // 计算一下当前要取的元素的下标.
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }

    public String getArguments() {
        //把当前的 arguments 参数从 Map 转成 String (Json)
        ObjectMapper objectMapper=new ObjectMapper();
        try {//有可能当前给的 Map 对象不一定完全遵守 Json 的格式
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";//如果真异常了就返回一个空的 Json 字符串即可
    }

    //从数据库中读数据时构造 Exchange 对象会自动调用到
    public void setArguments(String argumentsJson) {
        //把参数中的 argumentsJson 按照 Json 格式解析转换成 Map 对象
        ObjectMapper objectMapper=new ObjectMapper();
        //这里第二个参数是 通过这个把这个参数转换成 HashMap，也可以转换成其他 Java类对象
        //我们这里是集合类这样的复杂类型，可以使用 TypeReference 匿名内部类描述复杂类型的具体信息，通过泛型参数来描述
        try {
            //可能格式不对就会异常了
            this.arguments=objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String,Object>>() {});
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    //这两个方法也是用于 Java 测试
    public Object getArguments(String key){
        return arguments.get(key);
    }

    public void setArguments(String key,Object value) {
        arguments.put(key,value);
    }

    public void setArguments(Map<String, Object> arguments){
        this.arguments=arguments;
    }
}